第一步,下载安装CUDA
1 Tensorflow 1.8.0,对应CUDA9.0版。
2 安装前如果C:\Program Files下有NVIDIA Corporation等文件夹,先删除。
3 添加环境变量:1
2C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
4 注销后,重新登录让环境变量生效。
第二步,下载安装cudnn
1 这个相当于CUDA补丁,下载需要注册NVIDIA账号。
2 下载完后解压,拷贝到CUDA安装目录。
第三步,安装Tensorflow GPU版本
1 先卸载旧版,防止冲突。
2 然后,执行下面命令,下载最新版1.8.0 的Tensorflow1
pip install tensorflow-gpu